python -W ignore -u train.py \
--dist_port 6411 \
--train_dataset_root_path /home/workdir/lhb/coco2014 \
--lpips_coefficient 3.50 \
--joint_image_text_loss_coefficient 0.0025 \
--epochs 50 \
--aux-learning-rate 1e-3 \
--num-workers 8 \
--lambda 0.0004 \
--batch-size 8 \
--patch-size 256 256 \
--seed 100 \
--clip_max_norm 1.0 \
--lr_epoch 45 48 